package com.yohann.algorithm;

/**
 * <p>
 * 八皇后问题
 * </p>
 *
 * @author Yohann
 * @since 2020/9/24
 */
public class QueenTest {
    //皇后个数
    int max = 8;
    //皇后位置
    int[] array = new int[max];
    int count = 0;
    int judgeCount = 0;

    public static void main(String[] args) {
        QueenTest queen = new QueenTest();

        queen.check(0);
        System.out.println("解法：" + queen.count);
        System.out.println("判断次数：" + queen.judgeCount);
    }

    //放置皇后
    private void check(int n) {
        //已找到结果
        if (n == max) {
            print();
            return;
        }

        //依次放入皇后判断是否冲突
        for (int i = 0; i < max; i++) {
            //先把当前皇后放入第一列
            array[n] = i;

            //判断是否冲突 不冲突就递归 冲突就下一列
            if (judge(n)) {
                //开始放第n+1个皇后
                check(n + 1);
            }
        }
    }

    //查看放置的第n个皇后是否冲突
    private boolean judge(int n) {
        judgeCount++;
        for (int i = 0; i < n; i++) {
            //在同一列 纵坐标相同 或者 在同一斜线 即 k=1/-1 构成45度角
            if (array[i] == array[n] || Math.abs(n - i) == Math.abs(array[n] - array[i])) {
                return false;
            }
        }

        return true;
    }

    //将皇后位置输出
    private void print() {
        count++;
        for (int i = 0; i < array.length; i++) {
            System.out.print(array[i] + " ");
        }
        System.out.println();
        System.out.println();
    }
}